What is the correct way to code this in Swift